﻿#pragma once
#include<string>
#include<vector>
#include<iostream>
using namespace std;

namespace wzy
{
	// 请完成哈希表的如下操作
	// 哈希函数采用除留余数法﻿
	template<class K>
	struct HashFunc
	{
		size_t operator()(const K& key)
		{
			return (size_t)key;
		}
	};

	// 哈希表中支持字符串的操作
	template<>
	struct HashFunc<string>
	{
		size_t operator()(const string& key)
		{
			size_t hash = 0;
			for (auto e : key)
			{
				hash *= 31;
				hash += e;
			}

			return hash;
		}
	};

	// 以下采用开放定址法，即线性探测解决冲突
	namespace open_address
	{
		enum State
		{
			EXIST,
			EMPTY,
			DELETE
		};

		template<class K, class V>
		struct HashData
		{
			pair<K, V> _kv;
			State _state = EMPTY;
		};

		template<class K, class V, class Hash = HashFunc<K>>
		class HashTable
		{
		public:
			HashTable()
			{
				_tables.resize(10);
			}

			bool Insert(const pair<K, V>& kv)
			{
				if (Find(kv.first))
					return false;

				//扩容
				if (_n * 10 / _tables.size() >= 7)
				{
					HashTable newHT;
					newHT._tables.resize(_tables.size() * 2);
					for (size_t i = 0; i < _tables.size(); i++)
					{
						if (_tables[i]._state == EXIST)
							newHT.Insert(_tables[i]._kv);
					}
					_tables.swap(newHT._tables);
				}

				Hash hash;
				size_t hashi = hash(kv.first) % _tables.size();
				while (_tables[hashi]._state == EXIST)
				{
					++hashi;
					hashi %= _tables.size();
				}

				_tables[hashi]._kv = kv;
				_tables[hashi]._state = EXIST;
				++_n;
				return true;
			}



			HashData<K, V>* Find(const K& key)
			{
				Hash hash;
				size_t hashi = hash(key) % _tables.size();
				while (_tables[hashi]._state != EMPTY)
				{
					if (_tables[hashi]._state == EXIST
						&& _tables[hashi]._kv.first == key)
						return &_tables[hashi];
					
					++hashi;
					hashi %= _tables.size();
				}
				return nullptr;
			}

			bool Erase(const K& key)
			{
				HashData<K, V>* ret = Find(key);
				if (ret == nullptr)
					return false;
				else
				{
					ret->_state = DELETE;
					return true;
				}
			}


		private:
			vector<HashData<K, V>> _tables;
			size_t _n = 0;  // 表中存储数据个数
		};
	}


	// 实现完成后，请对整形 和 字符串完成测试
	void test_int()
	{
		open_address::HashTable<int, int> ht;
		int a[] = { 11,52,82,72,25,56,17};
		for (auto e : a)
		{
			ht.Insert({e,e});
		}

		ht.Insert({ 22,00 });
		ht.Insert({ 15,11 });
		ht.Insert({ 90,22 });

		cout << ht.Find(52) << endl;
		ht.Erase(52);
		cout << ht.Find(52) << endl;

		cout << ht.Find(72) << endl;
	}

	void test_string()
	{
		open_address::HashTable<string, string> ht;
		ht.Insert({ "insert","插入" });
		ht.Insert({ "delete","删除" });
		ht.Insert({ "modify","修改" });
		ht.Insert({ "add","增加" });

		cout << HashFunc<string>()("abcd") << endl;
		cout << HashFunc<string>()("adbc") << endl;
		cout << HashFunc<string>()("aadd") << endl;
		cout << HashFunc<string>()("adad") << endl;

	}

}